import numpy as np


def gen_golden_data_logaddexp():
    # 生成输入数据（范围调整为-50到50，避免数值溢出）
    input_x = np.random.uniform(-50, 50, [8, 2048]).astype(np.float32)
    input_y = np.random.uniform(-50, 50, [8, 2048]).astype(np.float32)
    
    # 计算logaddexp(x,y) = ln(e^x + e^y)
    golden = np.logaddexp(input_x, input_y).astype(np.float32)

    # 保存输入输出数据
    input_x.tofile("./input/input_x.bin")
    input_y.tofile("./input/input_y.bin")
    golden.tofile("./output/golden.bin")


if __name__ == "__main__":
    gen_golden_data_logaddexp()